﻿using System;
using System.IO;
using IWshRuntimeLibrary;
using Microsoft.Win32;
using NLog.Fluent;

namespace Labthink.PCSoft.UniversalHost4.Utility
{
    public class ShortcutAndAutoRunCreator
    {
        //需要引入IWshRuntimeLibrary，搜索Windows Script Host Object Model

        /// <summary>
        /// 创建快捷方式
        /// </summary>
        /// <param name="directory">快捷方式所处的文件夹</param>
        /// <param name="shortcutName">快捷方式名称</param>
        /// <param name="targetPath">目标路径</param>
        /// <param name="description">描述</param>
        /// <param name="iconLocation">图标路径，格式为"可执行文件或DLL路径, 图标编号"，
        /// 例如System.Environment.SystemDirectory + "\\" + "shell32.dll, 165"</param>
        /// <remarks></remarks>
        public static void CreateShortcut(string directory, string shortcutName, string targetPath,
            string description = null, string iconLocation = null)
        {
            if (!Directory.Exists(directory))
            {
                Directory.CreateDirectory(directory);
            }

            string shortcutPath = Path.Combine(directory, $"{shortcutName}.lnk");
            WshShell shell = new WshShell();
            IWshShortcut shortcut = (IWshShortcut)shell.CreateShortcut(shortcutPath);//创建快捷方式对象
            shortcut.TargetPath = targetPath;//指定目标路径
            shortcut.WorkingDirectory = Path.GetDirectoryName(targetPath);//设置起始位置
            shortcut.WindowStyle = 1;//设置运行方式，默认为常规窗口
            shortcut.Description = description;//设置备注
            shortcut.IconLocation = string.IsNullOrWhiteSpace(iconLocation) ? targetPath : iconLocation;//设置图标路径
            shortcut.Save();//保存快捷方式
        }

        /// <summary>
        /// 创建桌面快捷方式
        /// </summary>
        /// <param name="shortcutName">快捷方式名称</param>
        /// <param name="targetPath">目标路径</param>
        /// <param name="description">描述</param>
        /// <param name="iconLocation">图标路径，格式为"可执行文件或DLL路径, 图标编号"</param>
        /// <remarks></remarks>
        public static void CreateShortcutOnDesktop(string shortcutName, string targetPath,
            string description = null, string iconLocation = null)
        {
            string desktop = Environment.GetFolderPath(Environment.SpecialFolder.DesktopDirectory);//获取桌面文件夹路径
            CreateShortcut(desktop, shortcutName, targetPath, description, iconLocation);
        }

        /// <summary>
        /// 判断桌面是否包含快捷方式
        /// </summary>
        /// <param name="shortcutName">快捷方式名称</param>
        /// <returns></returns>
        public static bool ExistsShortcutOnDesktop(string shortcutName)
        {
            var desktop = Environment.GetFolderPath(Environment.SpecialFolder.DesktopDirectory);//获取桌面文件夹路径
            string shortcutPath = $"{shortcutName}.lnk" ;
            return System.IO.File.Exists(Path.Combine(desktop, shortcutPath));
        }

        /// <summary>
        /// 在注册表中设置自启动路径
        /// </summary>
        /// <param name="path">路径</param>
        /// <param name="name">名称</param>
        public static void SetAutoRun(string path,string name)
        {
            try
            {
                using (var loca = Registry.LocalMachine)
                {
                    using (var run = loca.CreateSubKey(@"SOFTWARE\Microsoft\Windows\CurrentVersion\Run"))
                    {
                        if (run.GetValue(name) == null)
                        {
                            run.SetValue(name, path);
                        }
                    }
                }
            }
            catch(Exception e)
            {
                Log.Error().Exception(e).Write();
            }
           
            //oca = Registry.LocalMachine;

        }
    }
}
